Searched defs:return_shape (Results 1 – 10 of 10) sorted by relevance
111 ShapedType return_shape = op->getResultTypes()[0]; in TryToCacheResultsTensorValue() local124 ShapedType return_shape = op->getResultTypes()[0]; in TryToCacheResultsTensorValue() local145 ShapedType return_shape = op->getResultTypes()[idx]; in TryToCacheResultsTensorValue() local
989 const Shape* return_shape = &shape; in GetSubshape() local1000 const Shape* return_shape = &shape; in TryGetSubshape() local1015 Shape* return_shape = shape; in GetMutableSubshape() local
137 const Shape& return_shape = root_instruction->shape(); in CodegenNestedComputation() local
630 const Shape& return_shape = computation.root_instruction()->shape(); in ComputeNestedElementFromAddrs() local
984 Shape return_shape; in TransferToClient() local
132 const Shape& return_shape = computation->root_instruction()->shape(); in EmitThreadLocalFunctionEpilogue() local3440 const Shape& return_shape = callee.root_instruction()->shape(); in EmitThreadLocalCall() local
1800 xla::Shape return_shape = xla::ShapeUtil::MakeTupleShape( in TEST() local1899 xla::Shape return_shape = xla::ShapeUtil::MakeTupleShape( in TEST() local
2177 xla::Shape return_shape = xla::TypeToShape(ret.get().getType()); in Lower() local
1757 SmallVector<int64_t, 4> return_shape(num_dims, ShapedType::kDynamicSize); in inferConvReturnTypeComponents() local
1249 SmallVector<int64_t, 4> return_shape(4, ShapedType::kDynamicSize); in inferReturnTypes() local