/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 | concat_op.cc | 76 const TensorShape& in_shape = shapes[i]; in Compile() local
|
D | lrn_ops.cc | 38 const TensorShape in_shape = ctx->InputShape(0); in Compile() local
|
/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 | 261 void BuildNodeDef(const TensorShape& in_shape, DataType dtype, in BuildNodeDef() 277 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 | 426 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | cudnn_determinism_test.py | 55 def _random_out_op(self, in_shape, filter_shape): argument
|
/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/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/compiler/tf2tensorrt/convert/ |
D | convert_graph.cc | 453 TensorShapeProto in_shape; in CreateTRTNode() local
|
D | convert_nodes.cc | 252 auto in_shape = input_params.at(in_port); in GetInputProperties() local
|
/external/tensorflow/tensorflow/core/nccl/ |
D | nccl_manager_test.cc | 138 TensorShape in_shape, TensorShape out_shape) { in MakeGatherTestCase()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_dataflow_analysis_test.cc | 2121 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 2264 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
D | tuple_points_to_analysis_test.cc | 931 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
D | algebraic_simplifier_test.cc | 3383 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 | 1653 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local 1727 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local 1833 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
|
/external/tensorflow/tensorflow/lite/toco/ |
D | tooling_util.cc | 2216 std::vector<int> in_shape = input_array.shape().dims(); in ReshapeIsEquivalentToTranspose() local
|