Home
last modified time | relevance | path

Searched defs:data_shape (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops.cc47 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
Drange_coder_ops_test.cc135 const TensorShape& data_shape = data.shape(); in TestEncodeAndDecode() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc199 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
Dreduction_ops_common.cc48 const TensorShape data_shape = ctx->InputShape(0); in Compile() local
Dsegment_reduction_ops.cc54 TensorShape data_shape = ctx->InputShape(0); in Compile() local
Ddynamic_stitch_op.cc75 const TensorShape& data_shape = data_shapes[input_num]; in Compile() local
/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad_test.cc50 TensorShape data_shape({2, 3, 2}); in TEST_F() local
Dimage_grad_test.cc43 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/
Dmath_ops.cc986 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
Ddata_flow_ops.cc58 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/
Dhlo_dataflow_analysis_test.cc1992 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 …]
Dtuple_points_to_analysis_test.cc876 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
Dbuffer_liveness_test.cc618 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in BuildModule() local
776 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in Run() local
Dlayout_assignment.cc403 const Shape& data_shape = in BuildHostChannelConstraints() local
Dhlo_instruction_test.cc1234 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
Dhlo_instruction.cc363 const Shape& data_shape = ShapeUtil::GetTupleElementShape(shape, 0); in CreateFromProto() local
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops.cc135 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/
Dxla.proto487 ShapeProto data_shape = 2; field
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtuple_test.cc433 Shape data_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc331 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed() local