Home
last modified time | relevance | path

Searched defs:shape_attrs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dprepare_tpu_computation_for_tf_export.cc68 llvm::SmallVector<Attribute> shape_attrs; in matchAndRewrite() local
Ddecompose_reduce_dataset.cc73 llvm::SmallVector<Attribute, 2> shape_attrs; in CreateIterator() local
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc602 Shape::Attrs shape_attrs; in TileGrad() local
705 Shape::Attrs shape_attrs; in GatherV2Grad() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.cc786 std::vector<const TensorShapeProto*> shape_attrs; in ValidateShape() local