Home
last modified time | relevance | path

Searched refs:output_shape_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_to_dense_op.cc231 Tensor output_shape_tensor; in ComputeAsync() local
235 {output_shape_vec.size()}, &output_shape_tensor), in ComputeAsync()
238 AsDeviceMemory(output_shape_tensor.template flat<Index>().data(), in ComputeAsync()
239 output_shape_tensor.template flat<Index>().size()); in ComputeAsync()
244 output_shape_tensor.NumElements() * sizeof(Index)) in ComputeAsync()
253 output_shape_tensor, default_value.scalar<T>()(), output); in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dsparse_ops.py2054 output_shape_tensor = math_ops.maximum(
2058 output_shape_tensor = ops.convert_to_tensor(new_shape)
2059 output_shape_tensor.get_shape().assert_has_rank(1)
2060 output_shape_tensor = math_ops.cast(output_shape_tensor, dtypes.int64)
2063 if output_shape_tensor.get_shape().rank is not None:
2064 output_shape_tensor.get_shape().dims[0].assert_is_compatible_with(
2067 output_shape_tensor_const = tensor_util.constant_value(output_shape_tensor)
2077 output_shape_tensor = output_shape_tensor_const
2080 output_shape_tensor = control_flow_ops.with_dependencies([
2082 array_ops.shape(in_shape), array_ops.shape(output_shape_tensor))
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A Dconvolutional.py1026 output_shape_tensor = array_ops.stack(output_shape)
1030 output_shape_tensor,
1318 output_shape_tensor = array_ops.stack(output_shape)
1322 output_shape_tensor,
1627 output_shape_tensor = array_ops.stack(output_shape)
1631 output_shape_tensor,
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc4582 const TfLiteTensor& output_shape_tensor = in VisitTransposeConvNode() local
4585 CheckTensorType(logging_context, output_shape_tensor, kTfLiteInt32, in VisitTransposeConvNode()
4588 CheckShapeTensorShape(logging_context, output_shape_tensor, in VisitTransposeConvNode()
4591 CheckTensorStaticAllocation(logging_context, output_shape_tensor, in VisitTransposeConvNode()
4593 const int output_shape_dims = SizeOfDimension(&output_shape_tensor, 0); in VisitTransposeConvNode()
4665 const int32_t* output_shape = GetTensorData<int32_t>(&output_shape_tensor); in VisitTransposeConvNode()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc2961 Tensor output_shape_tensor(DT_INT32, in TrySimplify() local
2964 output_shape_tensor.flat<int32>()(i) = output_shape.dim(i).size(); in TrySimplify()
2995 new_shape_node->name(), TensorValue(&output_shape_tensor), in TrySimplify()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc5634 ITensorProxyPtr output_shape_tensor; in ConvertResize() local
5686 output_shape_tensor = result.ValueOrDie(); in ConvertResize()
5706 layer->setInput(1, *output_shape_tensor->trt_tensor()); in ConvertResize()