Searched refs:axis_t (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 303 Tensor axis_t; in SparsifyGatherInternal() local
304 TF_RETURN_IF_ERROR(GetNodeAttr(axis_node, "value", &axis_t)); in SparsifyGatherInternal()
306 if (axis_t.dtype() == DT_INT32) { in SparsifyGatherInternal()
307 axis = axis_t.scalar<int32>()(); in SparsifyGatherInternal()
308 } else if (axis_t.dtype() == DT_INT64) { in SparsifyGatherInternal()
309 axis = axis_t.scalar<int64>()(); in SparsifyGatherInternal()
|
| D | sparsify_gather_test.cc | 64 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/ |
| D | array_ops.cc | 1124 const Tensor* axis_t = c->input_tensor(2); in __anon7c94107b1802() local
1128 if (axis_t == nullptr) { in __anon7c94107b1802()
1140 if (axis_t->dtype() == DT_INT32) { in __anon7c94107b1802()
1141 axis = axis_t->scalar<int32>()(); in __anon7c94107b1802()
1143 axis = axis_t->scalar<int64>()(); in __anon7c94107b1802()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2297 Tensor axis_t(DT_INT32, TensorShape({})); in SimplifyPack() local
2302 if (!SetTensorValue(DT_INT32, axis, &axis_t).ok() || in SimplifyPack()
2303 !CreateNodeDef(axis_node->name(), TensorValue(&axis_t), axis_node) in SimplifyPack()
|
| D | arithmetic_optimizer.cc | 3284 auto* axis_t = (*axis_attr)["value"].mutable_tensor(); in RewriteGraph() local
3285 axis_t->set_dtype(DT_INT32); in RewriteGraph()
3286 axis_t->add_int_val(pack_axis); in RewriteGraph()
|