| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | reshape_builder.cc | 30 void PopulateOutputShapeFromTensor(const TfLiteTensor* shape_tensor, in PopulateOutputShapeFromTensor()
61 TfLiteTensor* shape_tensor = nullptr; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | broadcast_to_op.cc | 47 const Tensor& shape_tensor = ctx->input(1); in Compute() local
|
| 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
|
| D | parameterized_truncated_normal_op.cc | 620 const Tensor& shape_tensor = ctx->input(0); in Compute() local
741 const Tensor& shape_tensor = ctx->input(0); in Compute() local
|
| D | sparse_conditional_accumulator.h | 442 Tensor* shape_tensor; in ReturnShapeTensor() local
|
| D | collective_ops.cc | 900 const Tensor& shape_tensor = c->input(3); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reshape_test.cc | 66 TfLiteTensor* shape_tensor, in TestReshapeWithShape()
125 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshape() local
147 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshapeQuantized() local
346 auto shape_tensor = tflite::testing::CreateTensor(shape_data, shape_dims); in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 56 Tensor shape_tensor(tensorflow::DT_INT32, in FuzzImpl() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 257 def shape_tensor(shape, name=None): function
|
| D | linear_operator.py | 321 def shape_tensor(self, name="shape_tensor"): member in LinearOperator
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 72 auto shape_tensor = in createReshapeOp() local
|
| D | einsum.cc | 79 auto shape_tensor = in createReshapeOp() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 1068 def shape_tensor(shape): # pylint: disable=invalid-name function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 104 const TfLiteTensor* shape_tensor, in ResizeTensor()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 391 Tensor shape_tensor(DT_INT32, TensorShape({})); in TestFoldFusedBatchNormsWithConcat() local
|
| D | quantize_nodes_test.cc | 371 Tensor shape_tensor(DT_INT32, TensorShape({})); in TestQuantizeConcat() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 1748 const Tensor* shape_tensor = c->input_tensor(2); in __anon725c8f641002() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 103 Value shape_tensor, int32_t val) { in CreateI32SplatTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 2564 const TfLiteTensor& shape_tensor = tensors[node->inputs->data[1]]; in VisitReshapeNode() local
2619 const TfLiteTensor& shape_tensor = tensors[node->inputs->data[1]]; in VisitResizeBilinearNode() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2419 const Tensor* shape_tensor = c->input_tensor(2); in SparseReduceShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 610 auto shape_tensor = builder->create<mhlo::ConstOp>( in CreateRngUniform32() local
2805 auto shape_tensor = rewriter->create<shape::ToExtentTensorOp>( in BroadcastBatchMatMulV2Operands() local
|