Searched defs:source_shape (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_thunk.cc | 46 const Shape& source_shape = in ExecuteOnStream() local
|
| D | jitrt_custom_calls.cc | 986 const Shape& source_shape = in operator ()() local
1064 Shape source_shape = ToShape(*source); in operator ()() local
|
| D | ir_emitter_unnested.cc | 1903 const Shape source_shape = GetShape(select_and_scatter_op.getSource()); in EmitSelectAndScatter() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | select_and_scatter_expander.cc | 46 const auto source_shape = source->shape(); in ExpandInstruction() local
|
| D | indexed_array_analysis.cc | 482 const Shape& source_shape = operand->source()->shape(); in ReshapeToRemoveDegenerateDims() local
|
| D | dynamic_dimension_inference_test.cc | 997 auto source_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
|
| D | hlo_sharding_util.cc | 355 std::optional<HloSharding> ReshapeSharding(const Shape& source_shape, in ReshapeSharding()
|
| D | shape_inference.cc | 2353 const Window& window, const Shape& source_shape, in InferSelectAndScatterShape()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_and_scatter_test.cc | 45 std::vector<int64_t> source_shape; member
80 auto source_shape = GetParam().source_shape; in XLA_TEST_P() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | dynamic_ragged_shape.py | 2092 source_shape, argument
2142 def source_shape(self): member in _Broadcaster
2267 def _get_broadcaster(source_shape, target_shape): argument
|
| D | dynamic_ragged_shape_test.py | 2397 def testBroadcasterInitRaises(self, source_shape, target_shape, argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 202 const Shape& source_shape, const Shape& dest_shape, in StrideConfig()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1490 auto source_shape = op.source().getType().cast<ShapedType>(); in InferShapeForXlaSelectAndScatterOp() local
|