Home
last modified time | relevance | path

Searched refs:kStack (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dconvert_trivial_stack_to_reshape.cc30 if (stack_it->get()->type != OperatorType::kStack) { in Run()
Dresolve_constant_stack.cc55 if (base_op->type != OperatorType::kStack) { in Run()
Dpropagate_fixed_sizes.cc1489 case OperatorType::kStack: in Run()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h36 kStack, enumerator
Dxla_resource.cc106 case kStack: { in SetZeroValue()
Dxla_compiler.cc267 case XlaResource::kStack: { in XLAShapeForArgument()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc119 ctx, xc.CreateResource(XlaResource::kStack, -1, std::move(name), dtype_, in Compile()
/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h76 kStack, enumerator
979 StackOperator() : Operator(OperatorType::kStack) {}
Dexport_tensorflow.cc1759 } else if (src_op.type == OperatorType::kStack) { in ConvertOperator()