| /external/tensorflow/tensorflow/core/framework/ |
| D | control_flow.h | 32 struct FrameAndIter { struct
36 FrameAndIter() {} in FrameAndIter() argument
38 FrameAndIter(uint64 frame, int64 iter) { in FrameAndIter() function
43 bool operator==(const FrameAndIter& other) const {
49 size_t operator()(const FrameAndIter& key) const { in operator()
51 CHECK_EQ(sizeof(uint64) + sizeof(int64), sizeof(FrameAndIter)); in operator()
52 return Hash64(reinterpret_cast<const char*>(&key), sizeof(FrameAndIter)); in operator()
|
| D | rendezvous.h | 176 const FrameAndIter& frame_iter);
|
| D | rendezvous.cc | 56 const FrameAndIter& frame_iter) { in CreateKey()
|
| D | rendezvous_test.cc | 44 "/job:mnist/replica:1/task:2/device:GPU:0", "var0", FrameAndIter(0, 0)); in TEST()
103 name, FrameAndIter(0, 0)); in MakeKey()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 39 const FrameAndIter& frame_iter, string* key) { in GetRendezvousKey()
45 static FrameAndIter GetFrameAndIter(OpKernelContext* ctx, in GetFrameAndIter()
52 return FrameAndIter(reinterpret_cast<uint64>(ctx->call_frame()), 0); in GetFrameAndIter()
93 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in Compute()
94 if (frame_iter == FrameAndIter(0, 0)) { in Compute()
201 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in ComputeAsync()
202 if (frame_iter == FrameAndIter(0, 0)) { in ComputeAsync()
|
| D | constant_op_test.cc | 71 params.frame_iter = FrameAndIter(0, 0); in PersistentMemoryTrackingTest()
|
| D | restore_op_test.cc | 176 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
421 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
|
| D | segment_reduction_ops_test.cc | 72 params.frame_iter = FrameAndIter(0, 0); in BM_SegmentReduction()
|
| D | parse_tensor_test.cc | 60 params.frame_iter = FrameAndIter(0, 0); in ParseSerializedWithNodeDef()
|
| D | variable_ops.cc | 33 const FrameAndIter& control_frame) { in TemporaryVariableName()
|
| D | sparse_to_dense_op_test.cc | 250 params.frame_iter = FrameAndIter(0, 0); in BM_SparseToDense()
|
| D | nn_ops_test.cc | 793 params.frame_iter = FrameAndIter(0, 0); in BM_LRNFloat()
872 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPool()
975 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPoolBk()
1066 params.frame_iter = FrameAndIter(0, 0); in BM_MaxPool()
1241 params.frame_iter = FrameAndIter(0, 0); in BM_ReluFloat()
1312 params.frame_iter = FrameAndIter(0, 0); in BM_SoftplusFloat()
|
| D | ops_testutil.cc | 168 params_->frame_iter = FrameAndIter(0, 0); in RunOpKernel()
|
| D | restore_v2_op_test.cc | 185 params.frame_iter = FrameAndIter(0, 0); in RunTest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 143 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
163 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
197 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
221 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
262 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
294 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
313 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_runner.cc | 137 tensor_name, FrameAndIter(0, 0)); in Run()
195 output_names[i], FrameAndIter(0, 0)); in Run()
|
| D | rendezvous_util_test.cc | 50 "/job:localhost/replica:0/task:0/device:GPU:0", name, FrameAndIter(0, 0)); in MakeStringKey()
|
| D | simple_propagator_state.h | 132 FrameAndIter GetFrameAndIter(const TaggedNode& tagged_node) const { in GetFrameAndIter()
|
| D | kernel_benchmark_testlib.cc | 158 recv_device, tensor_name, FrameAndIter(0, 0)); in GetRendezvousKey()
|
| D | propagator_state.h | 448 FrameAndIter GetFrameAndIter(const TaggedNode& tagged_node) const { in GetFrameAndIter()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 77 params.frame_iter = FrameAndIter(0, 0); in EvaluateNode()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 132 FrameAndIter(0, 0)); in ComputeAsync()
279 FrameAndIter(0, 0)); in Compute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.cc | 92 target, 1 /* src_incarnation */, target, in.name(), FrameAndIter(0, 0)); in ConstructFunctionGraph()
144 out.name(), FrameAndIter(0, 0)); in ConstructFunctionGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 250 params.frame_iter = FrameAndIter(0, 0); in Run()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 137 name, FrameAndIter(0, 0)), in Key()
|