Searched defs:elem_shape (Results 1 – 12 of 12) sorted by relevance
63 DataType dtype, const TensorShape& elem_shape) { in MaybeInitializeStack()135 TensorShape elem_shape = ctx->InputShape(1); in Compile() local
52 const TensorShape& elem_shape) { in MaybeInitializeTensorArray()200 TensorShape elem_shape = ctx->InputShape(2); in Compile() local369 TensorShape elem_shape = value_shape; in Compile() local507 TensorShape elem_shape = value_shape; in Compile() local
747 TF::TensorListElementShapeOp elem_shape, in HandleTensorListElementShapeOp()849 } else if (auto elem_shape = in DecomposeTensorListOpsInternal() local
95 llvm::SmallVector<int64_t, 8> elem_shape; in GetSplitElementTypeAndCount() local195 auto elem_shape = GetTensorArrayElementShape(ta, module); in HandleTensorArrayV3Op() local
647 const Shape& elem_shape = tuple_shapes[i]; in HumanString() local679 const Shape& elem_shape = tuple_shapes[i]; in HumanStringWithLayout() local
1229 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
498 auto elem_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1635 auto elem_shape = f32vec4_; in TEST_F() local1676 auto elem_shape = f32vec4_; in TEST_F() local
557 auto elem_shape = shape.tuple_shapes(tuple_index); in PropagateParameterLayoutToUsers() local
2078 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
821 auto elem_shape = rewriter->create<TF::SliceOp>( in CreateCondTrueBranch() local
2470 const Shape& elem_shape = in HandleCustomCall() local