Home
last modified time | relevance | path

Searched defs:const_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition.cc178 TensorShape const_shape; in PrepareSubgraphForFunctionConversion() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral.cc225 const Shape* const_shape = shape_.get(); in mutable_shape_do_not_use() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc601 ShapedType const_shape) { in IsValidConstShapeForMulConvPushDown()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc3576 const auto& const_shape = const_props[0].shape(); in MulConvPushDown() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc2661 llvm::SmallVector<int64_t, 4> const_shape; in build() local