Searched refs:Tvalue (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op.cc | 160 const Tensor& Tvalue = context->input(1); in Compute() local
161 OP_REQUIRES(context, IsLegacyScalar(Tvalue.shape()), in Compute()
163 Tvalue.shape().DebugString())); in Compute()
173 Tvalue.scalar<T>()); in Compute()
|
| D | inplace_ops.cc | 37 auto Tvalue = value.shaped<T, 2>({1, value.NumElements()}); in DoParallelConcatUpdate() local
41 Toutput.template chip<0>(r).device(d) = Tvalue.template chip<0>(0); in DoParallelConcatUpdate()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 65 template <class Tkey, class Tvalue>
66 bool EqualProtoMap(const ::tensorflow::protobuf::Map<Tkey, Tvalue>& a, in EqualProtoMap()
67 const ::tensorflow::protobuf::Map<Tkey, Tvalue>& b, in EqualProtoMap()
69 const std::function<string(const Tvalue&)>& value_to_string, in EqualProtoMap() argument
70 const std::function<bool(const Tkey&, const Tvalue&, in EqualProtoMap()
71 const Tvalue&)>& compare, in EqualProtoMap()
|