Searched refs:input_shape_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 31 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape() argument
34 Shape input_shape = ShapeUtil::MakeShape(primitive_type, input_shape_dims); in CreateModuleWithProgramShape()
46 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape() argument
49 Shape input_shape = ShapeUtil::MakeShape(primitive_type, input_shape_dims); in CreateModuleWithProgramShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1066 absl::Span<int64 const> input_shape_dims(shape.dimensions()); in CollapseFirstNDims() local
1068 std::accumulate(input_shape_dims.begin(), input_shape_dims.begin() + n, in CollapseFirstNDims()
1072 std::copy(input_shape_dims.begin() + n, input_shape_dims.end(), in CollapseFirstNDims()
|