| /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/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 181 Tensor shape_tensor(DT_INT32, shape); in CreateTRTEngine() local
365 Tensor shape_tensor(DT_INT32, shapes[1]); in TEST_P() 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
744 const Tensor& shape_tensor = ctx->input(0); in Compute() local
|
| D | sparse_conditional_accumulator.h | 417 Tensor* shape_tensor; in ReturnShapeTensor() 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 | 328 def shape_tensor(self, name="shape_tensor"): member in LinearOperator
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv3d_transpose.cc | 85 const TfLiteTensor* shape_tensor, const TfLiteTensor* filter, in ResizeOutputAndTemporaryTensors()
|
| D | transpose_conv.cc | 104 const TfLiteTensor* shape_tensor, in ResizeTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 80 auto shape_tensor = in createReshapeOp() local
|
| D | einsum.cc | 84 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/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 | 1783 const Tensor* shape_tensor = c->input_tensor(2); in __anon99f049891002() local
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 3089 const TfLiteTensor& shape_tensor = tensors[node->inputs->data[1]]; in VisitReshapeNode() local
3144 const TfLiteTensor& shape_tensor = tensors[node->inputs->data[1]]; in VisitResizeBilinearNode() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 132 Value shape_tensor, int32_t val) { in CreateI32SplatTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 101 Value shape_tensor = nullptr, in ExtractDynamicSizes()
1234 Value shape_tensor = args.size() > (isLHLO ? 1 : 0) ? args[0] : nullptr; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2459 const Tensor* shape_tensor = c->input_tensor(2); in SparseReduceShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 554 auto shape_tensor = builder->create<mhlo::ConstOp>( in CreateRngUniform32() local
3411 auto shape_tensor = rewriter->create<shape::ToExtentTensorOp>( in BroadcastBatchMatMulV2Operands() local
|