Home
last modified time | relevance | path

Searched refs:axis_t (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc302 Tensor axis_t; in SparsifyGatherInternal() local
303 TF_RETURN_IF_ERROR(GetNodeAttr(axis_node, "value", &axis_t)); in SparsifyGatherInternal()
305 if (axis_t.dtype() == DT_INT32) { in SparsifyGatherInternal()
306 axis = axis_t.scalar<int32>()(); in SparsifyGatherInternal()
307 } else if (axis_t.dtype() == DT_INT64) { in SparsifyGatherInternal()
308 axis = axis_t.scalar<int64>()(); in SparsifyGatherInternal()
Dsparsify_gather_test.cc64 Tensor axis_t(DT_INT32, TensorShape({})); in MakeGather() local
65 axis_t.scalar<int32>()() = 0; in MakeGather()
66 SetNodeTensorAttr<int32>("value", axis_t, axis_node); in MakeGather()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1192 const Tensor* axis_t = c->input_tensor(2); in __anon42d741191802() local
1196 if (axis_t == nullptr) { in __anon42d741191802()
1208 if (axis_t->dtype() == DT_INT32) { in __anon42d741191802()
1209 axis = axis_t->scalar<int32>()(); in __anon42d741191802()
1211 axis = axis_t->scalar<int64>()(); in __anon42d741191802()
1447 const Tensor* axis_t = c->input_tensor(1); in UniqueIdxShapeFn() local
1448 if (axis_t == nullptr || !c->RankKnown(input)) { in UniqueIdxShapeFn()
1457 int32 n = axis_t->NumElements(); in UniqueIdxShapeFn()
1466 if (axis_t->dtype() == DT_INT32) { in UniqueIdxShapeFn()
1467 axis = static_cast<int64>(axis_t->flat<int32>()(0)); in UniqueIdxShapeFn()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dunique_op_test.py151 axis_t = array_ops.placeholder(dtypes.int32, shape=None)
152 _, idx = gen_array_ops.unique_v2(x, axis=axis_t)
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3480 auto* axis_t = (*axis_attr)["value"].mutable_tensor(); in RewriteGraph() local
3481 axis_t->set_dtype(DT_INT32); in RewriteGraph()
3482 axis_t->add_int_val(pack_axis); in RewriteGraph()
Dconstant_folding.cc2436 Tensor axis_t(DT_INT32, TensorShape({})); in SimplifyPack() local
2440 if (!SetTensorValue(DT_INT32, axis, &axis_t).ok() || in SimplifyPack()
2441 !CreateNodeDef(axis_node_name, TensorValue(&axis_t), &new_node).ok()) { in SimplifyPack()