| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | canonicalize_all_gather_for_cse.cc | 35 const Shape& in_shape = inst->operand(0)->shape(); in IsAddingOnlyDegenerateDimensions() local
60 const Shape& in_shape = ag->operand(0)->shape(); in RunOnComputation() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bcast_ops.cc | 44 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
93 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
|
| D | lrn_ops.cc | 38 const TensorShape in_shape = ctx->InputShape(0); in Compile() local
|
| D | concat_op.cc | 76 const TensorShape& in_shape = shapes[i]; in Compile() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | extract_volume_patches_grad_test.py | 60 def testGradient(self, in_shape, ksizes, strides): argument
|
| D | cudnn_deterministic_base.py | 74 def _random_out_op(self, in_shape, filter_shape, strides, padding): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op_test.cc | 32 TensorShape in_shape(chunk_size); in MakeGraph() local
|
| D | split_v_op_test.cc | 60 TensorShape in_shape(total_size); in MakeGraph() local
|
| D | scoped_allocator_ops_test.cc | 262 void BuildNodeDef(const TensorShape& in_shape, DataType dtype, in BuildNodeDef()
278 void MakeOp(const TensorShape& in_shape, DataType dtype, const string& name, in MakeOp()
|
| D | cudnn_pooling_gpu.cc | 41 const auto in_shape = tensor_in.shape(); in Compute() local
|
| D | fractional_avg_pool_op.cc | 260 TensorShape in_shape; in Compute() local
|
| D | bias_op.cc | 394 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_trivial_transpose_to_reshape.cc | 27 std::vector<int> in_shape) { in TransposeAffectsMemoryOrder()
|
| D | merge_reshape_into_preceding_transpose.cc | 66 std::vector<int> in_shape = input_array.shape().dims(); in ReshapeToTranspose() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test_base.py | 134 in_shape, argument
237 def _gpuVsCpuCase(self, in_shape, out_shape, align_corners, argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | copy_test.cc | 245 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1}); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 166 TensorShape in_shape, TensorShape out_shape) { in MakeGatherTestCase()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis_test.cc | 2247 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2360 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 4144 auto in_shape = make_shape(in_dims, options.input_minor_to_major_layout); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1654 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
1728 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
1834 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2260 std::vector<int> in_shape = input_array.shape().dims(); in ReshapeIsEquivalentToTranspose() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 1688 const auto& in_shape = input->tensor.shape; in Parse() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 285 auto in_shape = input_params.at(in_port); in GetInputProperties() local
|