Searched defs:op_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 47 const auto& op_shape = operands_shapes[i]; in GetOperandTypes() local
129 Shape op_shape = operands_shapes[0]; in AggregateToTopKBuilder() local
225 for (auto& op_shape : operands_shapes) { in ApproxTopK() local
|
| D | tridiagonal.cc | 39 Status CheckSecondToLastDimension(const Shape& op_shape, int64_t rank, in CheckSecondToLastDimension()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | random_op_spmd_expander.cc | 47 const llvm::SmallVector<int64_t, 4>& op_shape, const Layout& layout, in ValidateShapeAndGetNewShape()
175 llvm::SmallVector<int64_t, 4> op_shape; in GetNewLocalShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | zero_sized_hlo_elimination_test.cc | 85 Shape op_shape = ShapeUtil::MakeShape(F32, {4, 0}); in TEST_F() local
|
| D | hlo_creation_utils.cc | 494 Shape op_shape = on_true->shape(); in MakeSelectHlo() local
|
| D | algebraic_simplifier.cc | 331 const Shape& bitcast_shape, const Shape& op_shape, in ValidateTilingOfBitcast()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | approx_topk_op.cc | 67 xla::Shape op_shape = ctx->InputXlaShape(0).value(); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 4365 mlir::lmhlo::FusionOp fusion, const Shape& op_shape, in NumInputsInvolveInOnlyElementwiseOps()
|