/external/tensorflow/tensorflow/contrib/coder/kernels/ |
D | range_coder_ops.cc | 47 BroadcastRange(T* data_pointer, gtl::ArraySlice<int64> data_shape, in BroadcastRange() 118 Status CheckCdfShape(const TensorShape& data_shape, in CheckCdfShape() 150 std::vector<int64> data_shape, cdf_shape; in Compute() local 183 gtl::ArraySlice<int64> data_shape, in RangeEncodeImpl() 242 std::vector<int64> data_shape, cdf_shape; in Compute() local
|
D | range_coder_ops_test.cc | 135 const TensorShape& data_shape = data.shape(); in TestEncodeAndDecode() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | resampler_ops.cc | 199 xla::Shape data_shape, XlaOp sample) { in BoundSamples() 252 xla::Shape data_shape) { in CalculateGradData() 362 xla::PrimitiveType data_type, xla::Shape data_shape) { in CalculateGradWarp() 488 TensorShape data_shape = ctx->InputShape("data"); in Compile() local 640 xla::Shape data_shape = TensorShapeToXLAShape(data_type, data_shape_tf); in Compile() local
|
D | reduction_ops_common.cc | 48 const TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
D | segment_reduction_ops.cc | 54 TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
D | dynamic_stitch_op.cc | 75 const TensorShape& data_shape = data_shapes[input_num]; in Compile() local
|
/external/tensorflow/tensorflow/cc/gradients/ |
D | data_flow_grad_test.cc | 50 TensorShape data_shape({2, 3, 2}); in TEST_F() local
|
D | image_grad_test.cc | 43 Tensor MakeData(const TensorShape& data_shape) { in MakeData() 188 Tensor MakeData(const TensorShape& data_shape) { in MakeData() 305 Tensor MakeData(const TensorShape& data_shape) { in MakeData()
|
/external/tensorflow/tensorflow/core/ops/ |
D | math_ops.cc | 986 ShapeHandle data_shape; in SegmentReductionShapeFn() local 1002 ShapeHandle data_shape; in SparseSegmentReductionShapeFn() local 1026 ShapeHandle data_shape; in SparseSegmentReductionGradShapeFn() local 1064 ShapeHandle data_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
|
D | data_flow_ops.cc | 58 ShapeHandle data_shape = c->input(0); in __anon4a7ac08d0202() local 105 ShapeHandle data_shape = c->input(i + num_partitions); in DynamicStitchShapeFunction() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_dataflow_analysis_test.cc | 1992 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 2028 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 2096 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 2119 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 2154 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 2179 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 2303 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 2340 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 2379 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 2498 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local [all …]
|
D | tuple_points_to_analysis_test.cc | 876 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 970 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 1005 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local 1119 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 1155 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local 1182 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
D | buffer_liveness_test.cc | 618 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in BuildModule() local 776 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in Run() local
|
D | layout_assignment.cc | 403 const Shape& data_shape = in BuildHostChannelConstraints() local
|
D | hlo_instruction_test.cc | 1234 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
D | hlo_instruction.cc | 363 const Shape& data_shape = ShapeUtil::GetTupleElementShape(shape, 0); in CreateFromProto() local
|
/external/tensorflow/tensorflow/contrib/resampler/kernels/ |
D | resampler_ops.cc | 135 const ::tensorflow::TensorShape& data_shape = data.shape(); in Compute() local 340 const ::tensorflow::TensorShape& data_shape = data.shape(); in Compute() local
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | xla.proto | 487 ShapeProto data_shape = 2; field
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | tuple_test.cc | 433 Shape data_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.cc | 331 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed() local
|