Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather_test.cc64 Tensor axis_t(DT_INT32, TensorShape({})); in MakeGather() local
H A Dsparsify_gather.cc302 Tensor axis_t; in SparsifyGatherInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc1211 const Tensor* axis_t = c->input_tensor(2); in __anon38bbb0e81802() local
1470 const Tensor* axis_t = c->input_tensor(1); in UniqueIdxShapeFn() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc2486 Tensor axis_t(DT_INT32, TensorShape({})); in SimplifyPack() local
H A Darithmetic_optimizer.cc4064 auto* axis_t = (*axis_attr)["value"].mutable_tensor(); in RewriteGraph() local