Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather_test.cc64 Tensor axis_t(DT_INT32, TensorShape({})); in MakeGather() local
Dsparsify_gather.cc303 Tensor axis_t; in SparsifyGatherInternal() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1124 const Tensor* axis_t = c->input_tensor(2); in __anondb9326b21802() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2297 Tensor axis_t(DT_INT32, TensorShape({})); in SimplifyPack() local
Darithmetic_optimizer.cc3284 auto* axis_t = (*axis_attr)["value"].mutable_tensor(); in RewriteGraph() local