Searched defs:elem_shape (Results 1 – 13 of 13) 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() local368 TensorShape elem_shape = value_shape; in Compile() local506 TensorShape elem_shape = value_shape; in Compile() local
757 TF::TensorListElementShapeOp elem_shape, in HandleTensorListElementShapeOp()859 } else if (auto elem_shape = in DecomposeTensorListOpsInternal() local
96 llvm::SmallVector<int64_t, 8> elem_shape; in GetSplitElementTypeAndCount() local196 auto elem_shape = GetTensorArrayElementShape(ta, module); in HandleTensorArrayV3Op() local
1019 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local1043 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1082 const auto& elem_shape = while_shape.tuple_shapes(i); in TryMergeInductionVariables() local
904 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
454 auto elem_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1558 auto elem_shape = f32vec4_; in TEST_F() local1599 auto elem_shape = f32vec4_; in TEST_F() local
2095 Shape elem_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
584 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanString() local608 for (const Shape& elem_shape : shape.tuple_shapes()) { in HumanStringWithLayout() local
659 auto elem_shape = rewriter->create<TF::SliceOp>( in CreateCondTrueBranch() local
2338 const Shape& elem_shape = in HandleCustomCall() local