| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | reshape_builder.cc | 30 void PopulateOutputShapeFromTensor(const TfLiteTensor* shape_tensor, in PopulateOutputShapeFromTensor()
61 TfLiteTensor* shape_tensor = nullptr; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_resource_ops_test.cc | 187 Tensor shape_tensor(DT_INT32, shape); in CreateTRTEngine() local
377 Tensor shape_tensor(DT_INT32, shapes[1]); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | random_binomial_op.cc | 338 const Tensor& shape_tensor = ctx->input(2); in Compute() local
450 const Tensor& shape_tensor = ctx->input(0); in Compute() local
|
| H A D | parameterized_truncated_normal_op.cc | 621 const Tensor& shape_tensor = ctx->input(0); in Compute() local
743 const Tensor& shape_tensor = ctx->input(0); in Compute() local
|
| H A D | broadcast_to_op.cc | 47 const Tensor& shape_tensor = ctx->input(1); in Compute() local
|
| H A D | sparse_conditional_accumulator.h | 417 Tensor* shape_tensor; in ReturnShapeTensor() local
|
| H A D | set_kernels.cc | 73 const Tensor& shape_tensor = ctx->input(base_index + 2); in SparseTensorFromContext() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| H A D | scatter_nd_fuzz.cc | 56 Tensor shape_tensor(tensorflow::DT_INT32, in FuzzImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| H A D | multiplex_3_kernel.cc | 179 Status ValidateSparseTensor(const ::tensorflow::Tensor& indices_tensor, in ValidateSparseTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linear_operator_util.py | 254 def shape_tensor(shape, name=None): function
|
| H A D | linear_operator.py | 325 def shape_tensor(self, name="shape_tensor"): member in LinearOperator
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | unroll_batch_matmul.cc | 88 auto shape_tensor = in createReshapeOp() local
|
| H A D | einsum.cc | 103 auto shape_tensor = createI64ConstantOp(shape, loc, rewriter); in createReshapeOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | conv3d_transpose.cc | 85 const TfLiteTensor* shape_tensor, const TfLiteTensor* filter, in ResizeOutputAndTemporaryTensors()
|
| H A D | transpose_conv.cc | 104 const TfLiteTensor* shape_tensor, in ResizeTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | tensor_util.py | 1067 def shape_tensor(shape): # pylint: disable=invalid-name function
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_old_batch_norms_test.cc | 391 Tensor shape_tensor(DT_INT32, TensorShape({})); in TestFoldFusedBatchNormsWithConcat() local
|
| H A D | quantize_nodes_test.cc | 371 Tensor shape_tensor(DT_INT32, TensorShape({})); in TestQuantizeConcat() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | math_ops.cc | 1819 const Tensor* shape_tensor = c->input_tensor(2); in __anon247ac1531002() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 4163 const TfLiteTensor& shape_tensor = tensors[node->inputs->data[1]]; in VisitReshapeNode() local
4220 const TfLiteTensor& shape_tensor = tensors[node->inputs->data[1]]; in VisitResizeBilinearNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | lower_static_tensor_list.cc | 124 Value shape_tensor, int32_t val) { in CreateI32SplatTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device.cc | 953 std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> shape_tensor( in SparsePack() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 2511 const Tensor* shape_tensor = c->input_tensor(2); in SparseReduceShapeFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 484 auto shape_tensor = builder->create<mhlo::ConstantOp>( in CreateRngUniform32() local
3209 auto shape_tensor = rewriter->create<shape::ToExtentTensorOp>( in BroadcastBatchMatMulV2Operands() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 356 Tensor _reshape_from_tensor(const Tensor& self, const Tensor& shape_tensor) { in _reshape_from_tensor()
|