Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc57 const Shape& data_shape, const Shape& state_shape, in GetGeneratorComputation()
110 const Shape& data_shape = rng->shape().tuple_shapes(1); in ExpandInstruction() local
Drng_bit_generator_expander.h42 Shape data_shape; member
Dhlo_dataflow_analysis_test.cc2103 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2140 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2211 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2235 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2271 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2297 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
2396 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2434 Shape data_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2474 Shape data_shape = ShapeUtil::MakeShape(F32, {1, 8}); in TEST_F() local
2676 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
[all …]
Dwhile_loop_concat_code_motion.cc871 Shape data_shape = hlo->operand(i)->shape(); in RewriteLoopWithConcatGroups() local
Dlayout_assignment.cc515 const Shape& data_shape = in BuildHostChannelConstraints() local
Dhlo_instruction_test.cc1452 Shape data_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
Dtensor_utils.cc21 Status QuantizationAxisAndShapeValid(const TensorShape& data_shape, in QuantizationAxisAndShapeValid()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc202 xla::Shape data_shape, XlaOp sample) { in BoundSamples()
255 xla::Shape data_shape) { in CalculateGradData()
365 xla::PrimitiveType data_type, xla::Shape data_shape) { in CalculateGradWarp()
490 TensorShape data_shape = ctx->InputShape("data"); in Compile() local
638 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
Ddynamic_stitch_op.cc75 TensorShape& data_shape = data_shapes[input_num]; in Compile() local
Dsegment_reduction_ops.cc55 TensorShape data_shape = ctx->InputShape(0); in Compile() local
Ddynamic_partition_op.cc111 xla::Shape data_shape = ctx->InputXlaShape(0).value(); in Compile() local
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dget_item_op.cc36 const TensorShape& data_shape = ctx->InputShape(0); in Compile() local
Dsegment_reduction_ops.cc45 TensorShape data_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1156 ShapeHandle data_shape; in SegmentReductionShapeFn() local
1172 ShapeHandle data_shape; in SparseSegmentReductionShapeFn() local
1196 ShapeHandle data_shape; in SparseSegmentReductionGradShapeFn() local
1234 ShapeHandle data_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
Ddata_flow_ops.cc58 ShapeHandle data_shape = c->input(0); in __anon89875fc90202() local
105 ShapeHandle data_shape = c->input(i + num_partitions); in DynamicStitchShapeFunction() local
/external/tensorflow/tensorflow/cc/gradients/
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()
Ddata_flow_grad_test.cc50 TensorShape data_shape({2, 3, 2}); in TEST_F() local
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc209 TensorShape data_shape({1, 1}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtuple_test.cc273 Shape data_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto699 ShapeProto data_shape = 2; field
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1177 xla::Shape data_shape = xla::ShapeUtil::MakeTupleShape(subshapes); in ExportXlaOp() local
1290 xla::Shape data_shape; in ExportXlaOp() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc408 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed() local
2323 const Shape& data_shape = ShapeUtil::GetSubshape(hlo->shape(), {0}); in HandlePadToStatic() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc842 const Shape& data_shape = GetShape(pad_to_static.getOutput().front()); in EmitPadToStatic() local
872 const Shape& data_shape = GetShape(slice_to_dynamic.getOutput().front()); in EmitSliceToDynamic() local
2229 const Shape& data_shape = GetShape(scatter_op.getUpdates()); in EmitScatter() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc2231 ArrayRef<int64_t> data_shape = data_ty.getShape(); in verify() local

12