| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | rng_bit_generator_expander.cc | 57 const Shape& data_shape, const Shape& state_shape, in GetGeneratorComputation()
110 const Shape& data_shape = rng->shape().tuple_shapes(1); in ExpandInstruction() local
|
| D | rng_bit_generator_expander.h | 42 Shape data_shape; member
|
| D | hlo_dataflow_analysis_test.cc | 2118 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2154 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2222 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2245 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2280 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2305 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2399 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2436 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2475 Shape data_shape = ShapeUtil::MakeShape(F32, {1, 8}); in TEST_F() local
2615 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
[all …]
|
| D | tuple_points_to_analysis_test.cc | 651 Shape data_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
927 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| D | layout_assignment.cc | 413 const Shape& data_shape = in BuildHostChannelConstraints() local
|
| D | hlo_instruction_test.cc | 1353 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
|
| D | hlo_instruction.cc | 400 const Shape& data_shape = ShapeUtil::GetTupleElementShape(shape, 0); in CreateFromProto() 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 | dynamic_stitch_op.cc | 75 const TensorShape& data_shape = data_shapes[input_num]; in Compile() local
|
| D | segment_reduction_ops.cc | 54 TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | get_item_op.cc | 36 const TensorShape& data_shape = ctx->InputShape(0); in Compile() local
|
| D | segment_reduction_ops.cc | 45 TensorShape data_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 1123 ShapeHandle data_shape; in SegmentReductionShapeFn() local
1139 ShapeHandle data_shape; in SparseSegmentReductionShapeFn() local
1163 ShapeHandle data_shape; in SparseSegmentReductionGradShapeFn() local
1201 ShapeHandle data_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
|
| D | data_flow_ops.cc | 58 ShapeHandle data_shape = c->input(0); in __anonea01102b0202() local
105 ShapeHandle data_shape = c->input(i + num_partitions); in DynamicStitchShapeFunction() 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/python/keras/engine/ |
| D | training_integration_test.py | 149 def _run_fit_eval_predict(self, layer_to_test, input_shape, data_shape, argument
|
| /external/tensorflow/tensorflow/cc/client/ |
| D | client_session_test.cc | 209 TensorShape data_shape({1, 1}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 565 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 | 395 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed() local
2195 const Shape& data_shape = ShapeUtil::GetSubshape(hlo->shape(), {0}); in HandlePadToStatic() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 834 const Shape& data_shape = in EmitPadToStaticFromMlir() local
956 const Shape& data_shape = in EmitSliceToDynamicFromMlir() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1978 ArrayRef<int64_t> data_shape = data_ty.getShape(); in Verify() local
|